# URL security detection

A lightweight BERT model, specifically fine-tuned for URL classification tasks, capable of categorizing URLs into four types: benign, phishing, malware, and tampered.

This model is built based on DistilBERT. After fine-tuning, it can perform multi-label classification on emails and URLs to determine whether they are safe or have potential phishing risks.